package cn.tedu.v5;

import java.io.*;
import java.net.*;
import java.util.*;


/**
 * Version 5多人聊天+qq聊天+消息转发
 * 功能： 登录 + 聊天  【退出】
 * 登录步骤：1.私有化属性，保存账号与密码--userDatBase
 *         2.构造器，创建两个用户名与账号
 *         3.处理多个客户端，加私有化属性：账号、密码
 */
public class Server {
    private ServerSocket serverSocket;
    private static final  Integer PORT=4545;
    private List<PrintWriter> allOut=new ArrayList<>();
    //1.私有化属性--保存账号与密码
    private Map<String,String> userDataBase;
    public Server(){
        try {
            serverSocket=new ServerSocket(PORT);
            //2.Server对象启动--main方法--
            // 创建服务器之初，完成对象账号与密码赋值
            userDataBase=new HashMap<>();
            userDataBase.put("宇航员","123456");
            userDataBase.put("丁丁","123456");
        } catch(BindException e) {
            System.out.println("端口号重复创建....");
        }catch (IOException e) {
            System.out.println("申请端口号失败");
        }
    }

    public void start(){
        try {
            while(true){
                        System.out.println("----等待客户端连接----");
                        Socket socket = serverSocket.accept();

                    new Thread(new ClientHandler(socket)).start();
                    }
        } catch (IOException e) {
            System.out.println("服务器端通讯断开了");
        }finally {
            if(serverSocket!=null&&serverSocket.isClosed()){
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
            }

        }
    }

    public static void main(String[] args)  {
        Server server=new Server();
        server.start();
    }

    private synchronized void broadcastMessage(String message){

        for (PrintWriter pw:allOut){
            pw.println("消息广播："+message);
        }
    }
    private class ClientHandler implements Runnable{
        private Socket socket;
        private String ip;
        private BufferedReader br;
        private PrintWriter pw;
        //3.很多客户端都具备账号密码
        private String username;
        private String password;

        public ClientHandler(Socket s){
            this.socket=s;
            ip=socket.getInetAddress().getHostAddress();

            try {
                 br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
                pw=new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(),"UTF8")),true);

                synchronized (allOut){
                    allOut.add(pw);
                }

            } catch (IOException e) {
                System.out.println("客户端断开通讯~~");
            }
        }
        //4.4.1封装判断账号与密码是否正确的方法--Login
        private boolean Login(String user,String pass){
            //Map中是否存在这个账号和密码
            //return userDataBase.containsKey(user) 判断这个账号密码是否正确
            return userDataBase.containsKey(user)&&userDataBase.get(user).equals(pass);
        }
        //5.账号与密码验证成功---登录---聊天：读写功能--封装方法handlerChat()
 private void handlerChat(){
            BufferedReader Br = br;
            new Thread(()->{
                try {
                    String message;
                    while((message= Br.readLine())!=null){
                        System.out.println("客户端说："+message);
                        System.out.println("服务器端请输入：");
                        broadcastMessage(message);
                    }
                } catch (IOException e) {
                    System.out.println("客户端读入失败！");
                }
            }).start();
            Scanner input =new Scanner(System.in);
            while (true) {
                System.out.println("服务器端请输入：");
                String say= input.nextLine();

                pw.println("服务器说："+say);
            }
        }
        public void run() {
            try {
        //4.目的：启动客户端与服务器连接
        //  假设：一个客户端连接上了，服务器端应该怎末处理
        //4.1    登录验证   【服务器端与客户端交互问题】
          pw.println("请输入用户名：");//发给客户端
          username=br.readLine();// 读入客户端发来的账号
        //4.2  密码验证
          pw.println("请输入密码");//发给客户端
          password=br.readLine();//读入客户端输入的密码
        //4.3  判断账号与密码是否正确--账号很多情况
        //4.4  账号很多情况：封装方法--功能：判断账号与密码是否正确
        if (Login(username,password)) {  //判断Login 为true 账号与密码正确
        pw.println("登陆成功，欢迎进入聊天室");
        //聊天 4.5 封装方法--登陆成功--开始聊天：读写功能--handlerChat
         handlerChat();
        }else{//不通过
            pw.println("登陆失败，用户名或者密码错误！");
            socket.close(); //登录失败，关闭连接
        }


            } catch(NullPointerException e){
                System.out.println("IO流传输对象，" +
                        "有对象为null，立马核对，" +
                        "建议：查看try块中的对象赋值为null");
            } catch(IOException e){
                System.out.println("服务器端断开连接");
            } finally {
                synchronized (allOut) {
                    allOut.remove(pw);
                }
                if(br!=null)
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("结束通讯：关闭失败！");
                    }
                if (pw!=null) pw.close();
                if(socket!=null&&!socket.isClosed())
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败~");
                    }
            }
        }
    }

}